package com.zis.util;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang.StringUtils;
import org.springframework.util.Assert;

/**
 * 
 * <b>说明：</b>Cookie 辅助类
 * 
 * @ClassName: CookieUtils
 * @author zhaohaitao(2543) 日期： 2015-7-24 上午11:40:22
 * 
 */
public class CookieUtils
{
    
    /**
     * 获得cookie
     * 
     * @param request HttpServletRequest
     * @param name cookie name
     * @return if exist return cookie, else return null.
     */
    public static Cookie getCookie(HttpServletRequest request, String name)
    {
        Assert.notNull(request);
        Cookie[] cookies = request.getCookies();
        if (cookies != null && cookies.length > 0)
        {
            for (Cookie c : cookies)
            {
                if (c.getName().equals(name))
                {
                    return c;
                }
            }
        }
        return null;
    }
    
    /**
     * 根据部署路径，将cookie保存在根目录。
     * 
     * @param request
     * @param response
     * @param name
     * @param value
     * @param expiry
     * @param domain
     * @return
     */
    public static Cookie addCookie(HttpServletRequest request, HttpServletResponse response, String name, String value,
        Integer expiry, String domain)
    {
        Cookie cookie = new Cookie(name, value);
        if (expiry != null)
        {
            cookie.setMaxAge(expiry);
        }
        if (StringUtils.isNotBlank(domain))
        {
            cookie.setDomain(domain);
        }
        String ctx = request.getContextPath();
        cookie.setPath(StringUtils.isBlank(ctx) ? "/" : ctx);
        response.addCookie(cookie);
        return cookie;
    }
    
    /**
     * 取消cookie
     * 
     * @param request
     * @param response
     * @param name
     * @param domain
     */
    public static void cancleCookie(HttpServletRequest request, HttpServletResponse response, String name, String domain)
    {
        Cookie cookie = new Cookie(name, "");
        cookie.setMaxAge(0);
        String ctx = request.getContextPath();
        cookie.setPath(StringUtils.isBlank(ctx) ? "/" : ctx);
        if (StringUtils.isNotBlank(domain))
        {
            cookie.setDomain(domain);
        }
        response.addCookie(cookie);
    }
}
